package com.itbac.mybatisplusspringbootdemo.injector;

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.itbac.mybatisplusspringbootdemo.injector.methods.InsertIgnoreBatch;
import com.itbac.mybatisplusspringbootdemo.injector.methods.getLimitOne;

import java.util.ArrayList;
import java.util.List;

/**
 * SQL注入器
 * 如果继承 AbstractSqlInjector ，原来的方法也要写上，否则无法使用了。
 * 继承 DefaultSqlInjector ，就可以获取到里面的数据。 super.getMethodList()
 */
public class MySqlInjector extends DefaultSqlInjector {
    @Override
    public List<AbstractMethod> getMethodList() {
        List<AbstractMethod> list = new ArrayList<>(super.getMethodList());
        list.add(new getLimitOne());
        list.add(new InsertIgnoreBatch());
        return list;
    }
}
